
from config import Config
from garbage_dataset import GarbageDataSet


class DataManager:
    def __init__(self):
        g1 = GarbageDataSet()
        # x_train, y_train, x_test, y_test = g1.load_data("data/data" + str(Config.IMAGE_HEIGHT) + ".npy")
        # x_train, y_train, x_test, y_test = g1.load_data("data/data" + str(128) + ".npy")
        # g1.normalization(x_train, y_train, x_test, y_test)

        self.x_train, self.y_train, self.x_test, self.y_test = g1.load_data_from_npy()


data_manager = DataManager()